package com.yxq.mdt.app.utils;

import com.tencent.mm.opensdk.modelpay.PayReq;
import com.yxq.mdt.app.constant.Constant;
import com.yxq.mdt.app.modules.order.bean.OrderPayResultWxBean;
import com.yxq.mdt.app.modules.personal.bean.RechargeWxPayResultBean;
import com.yxq.mdt.app.modules.webview.bean.WxPayFromBean;

import java.security.MessageDigest;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;
import java.util.SortedMap;
import java.util.TreeMap;

/**
 * @author L
 * @time
 * @describe 微信支付 签名工具
 */
public class WXUtil {


    /**
     * 获取sign签名
     *
     * @return
     */
    public static String genPayReq(PayReq req, OrderPayResultWxBean request) {

        // 把参数的值传进去SortedMap集合里面
        SortedMap<Object, Object> parameters = new TreeMap<Object, Object>();

        parameters.put("appid", Constant.WECHAT_ID);
        parameters.put("timestamp", request.getTimeStamp());
        parameters.put("noncestr", request.getRes().getNonce_str());
        parameters.put("package", "Sign=WXPay");
        parameters.put("partnerid", Constant.WECHAT_MCH_ID);
        parameters.put("prepayid", request.getRes().getPrepay_id());

        String characterEncoding = "UTF-8";
        String mySign = createSign(characterEncoding, parameters);
        return mySign;
    }
    /**
     * 获取sign签名
     *
     * @return
     */
    public static String genPayReq(PayReq req, WxPayFromBean request) {

        // 把参数的值传进去SortedMap集合里面
        SortedMap<Object, Object> parameters = new TreeMap<Object, Object>();

        parameters.put("appid", Constant.WECHAT_ID);
        parameters.put("timestamp", request.getCanshu().getTimeStamp());
        parameters.put("noncestr", request.getCanshu().getNonceStr());
        parameters.put("package", "Sign=WXPay");
        parameters.put("partnerid", Constant.WECHAT_MCH_ID);
        parameters.put("prepayid", request.getCanshu().getPackageX().replaceAll("prepay_id=", ""));;

        String characterEncoding = "UTF-8";
        String mySign = createSign(characterEncoding, parameters);
        return mySign;
    }
    /**
     * 获取sign签名
     *
     * @return
     */
    public static String genPayReq(PayReq req, RechargeWxPayResultBean request) {

        // 把参数的值传进去SortedMap集合里面
        SortedMap<Object, Object> parameters = new TreeMap<Object, Object>();

        parameters.put("appid", Constant.WECHAT_ID);
        parameters.put("timestamp", request.getTimeStamp());
        parameters.put("noncestr", request.getNonceStr());
        parameters.put("package", "Sign=WXPay");
        parameters.put("partnerid", Constant.WECHAT_MCH_ID);
        parameters.put("prepayid", request.getPackageX().replaceAll("prepay_id=", ""));

        String characterEncoding = "UTF-8";
        String mySign = createSign(characterEncoding, parameters);
        return mySign;
    }

    /**
     * 微信支付签名算法sign
     *
     * @param characterEncoding
     * @param parameters
     * @return
     */
    @SuppressWarnings("unchecked")
    private static String createSign(String characterEncoding, SortedMap<Object, Object> parameters) {
        StringBuffer sb = new StringBuffer();
        Set es = parameters.entrySet();//所有参与传参的参数按照ASCII排序（升序）
        Iterator it = es.iterator();
        while (it.hasNext()) {
            Map.Entry entry = (Map.Entry) it.next();
            String k = (String) entry.getKey();
            Object v = entry.getValue();
            if (null != v && !"".equals(v)
                    && !"sign".equals(k) && !"key".equals(k)) {
                sb.append(k + "=" + v + "&");
            }
        }
        sb.append("key=" + Constant.WECHAT_KEY);
        String sign = MD5Encode(sb.toString(), characterEncoding).toUpperCase();
        return sign;
    }

    private static String byteArrayToHexString(byte b[]) {
        StringBuffer resultSb = new StringBuffer();
        for (int i = 0; i < b.length; i++)
            resultSb.append(byteToHexString(b[i]));

        return resultSb.toString();
    }

    private static String byteToHexString(byte b) {
        int n = b;
        if (n < 0)
            n += 256;
        int d1 = n / 16;
        int d2 = n % 16;
        return hexDigits[d1] + hexDigits[d2];
    }

    private static String MD5Encode(String origin, String charsetname) {
        String resultString = null;
        try {
            resultString = new String(origin);
            MessageDigest md = MessageDigest.getInstance("MD5");
            if (charsetname == null || "".equals(charsetname))
                resultString = byteArrayToHexString(md.digest(resultString
                        .getBytes()));
            else
                resultString = byteArrayToHexString(md.digest(resultString
                        .getBytes(charsetname)));
        } catch (Exception exception) {
        }
        return resultString;
    }

    private static final String hexDigits[] = {"0", "1", "2", "3", "4", "5",
            "6", "7", "8", "9", "a", "b", "c", "d", "e", "f"};
}